import { nameof } from "@ts-morph/common";
import { expect } from "chai";
import { FunctionDeclaration, TypeParameterVariance } from "../../../../compiler";
import {
  FunctionDeclarationOverloadStructure,
  FunctionDeclarationSpecificStructure,
  FunctionDeclarationStructure,
  OptionalKind,
  StructureKind,
} from "../../../../structures";
import { MakeRequired, OptionalProperties } from "../../../../typings";
import { fillStructures, getInfoFromText, OptionalKindAndTrivia, OptionalTrivia } from "../../testHelpers";

describe("FunctionDeclaration", () => {
  describe(nameof<FunctionDeclaration>("getName"), () => {
    function doTest(startCode: string, name: string | undefined) {
      const { firstChild } = getInfoFromText<FunctionDeclaration>(startCode);
      expect(firstChild.getName()).to.equal(name);
    }

    it("should be undefined when it doesn't have a name", () => {
      doTest("export default function() {}", undefined);
    });

    it("should get when has a name", () => {
      doTest("export default function name() {}", "name");
    });
  });

  describe(nameof<FunctionDeclaration>("insertOverloads"), () => {
    function doTest(startCode: string, index: number, structures: OptionalKind<FunctionDeclarationOverloadStructure>[], expectedCode: string) {
      const { firstChild, sourceFile } = getInfoFromText<FunctionDeclaration>(startCode);
      const result = firstChild.insertOverloads(index, structures);
      expect(result.length).to.equal(structures.length);
      expect(sourceFile.getFullText()).to.equal(expectedCode);
    }

    it("should insert when no other overloads exist", () => {
      doTest(
        "function identifier() {}\n",
        0,
        [{ returnType: "number" }, {}],
        "function identifier(): number;\nfunction identifier();\nfunction identifier() {}\n",
      );
    });

    it("should insert with the ambientable and exportable nodes the same as the implementation signature unless overwritten", () => {
      doTest(
        "declare async function* identifier(param: string): string {}\n",
        0,
        [{ returnType: "number", hasDeclareKeyword: false }, {}],
        "function identifier(): number;\ndeclare function identifier();\ndeclare async function* identifier(param: string): string {}\n",
      );
    });

    it("should be able to insert at start when another overload exists", () => {
      doTest(
        "function identifier();\nfunction identifier() {}\n",
        0,
        [{ returnType: "string" }],
        "function identifier(): string;\nfunction identifier();\nfunction identifier() {}\n",
      );
    });

    it("should be able to insert at end when another overload exists", () => {
      doTest(
        "function identifier();\nfunction identifier() {}\n",
        1,
        [{ returnType: "string" }],
        "function identifier();\nfunction identifier(): string;\nfunction identifier() {}\n",
      );
    });

    it("should be able to insert in the middle when other overloads exists", () => {
      doTest(
        "function identifier();\nfunction identifier();\nfunction identifier() {}\n",
        1,
        [{ returnType: "string" }],
        "function identifier();\nfunction identifier(): string;\nfunction identifier();\nfunction identifier() {}\n",
      );
    });
  });

  describe(nameof<FunctionDeclaration>("insertOverload"), () => {
    function doTest(startCode: string, index: number, structure: OptionalKind<FunctionDeclarationOverloadStructure>, expectedCode: string) {
      const { firstChild, sourceFile } = getInfoFromText<FunctionDeclaration>(startCode);
      const result = firstChild.insertOverload(index, structure);
      expect(result).to.be.instanceOf(FunctionDeclaration);
      expect(sourceFile.getFullText()).to.equal(expectedCode);
    }

    it("should insert", () => {
      doTest(
        "function identifier();\nfunction identifier();\nfunction identifier() {}\n",
        1,
        { returnType: "number" },
        "function identifier();\nfunction identifier(): number;\nfunction identifier();\nfunction identifier() {}\n",
      );
    });
  });

  describe(nameof<FunctionDeclaration>("addOverloads"), () => {
    function doTest(startCode: string, structures: OptionalKind<FunctionDeclarationOverloadStructure>[], expectedCode: string) {
      const { firstChild, sourceFile } = getInfoFromText<FunctionDeclaration>(startCode);
      const result = firstChild.addOverloads(structures);
      expect(result.length).to.equal(structures.length);
      expect(sourceFile.getFullText()).to.equal(expectedCode);
    }

    it("should add at the end", () => {
      doTest(
        "function identifier();\nfunction identifier() {}\n",
        [{ returnType: "number" }, { returnType: "string" }],
        "function identifier();\nfunction identifier(): number;\nfunction identifier(): string;\nfunction identifier() {}\n",
      );
    });
  });

  describe(nameof<FunctionDeclaration>("addOverload"), () => {
    function doTest(startCode: string, structure: OptionalKind<FunctionDeclarationOverloadStructure>, expectedCode: string) {
      const { firstChild, sourceFile } = getInfoFromText<FunctionDeclaration>(startCode);
      const result = firstChild.addOverload(structure);
      expect(result).to.be.instanceOf(FunctionDeclaration);
      expect(sourceFile.getFullText()).to.equal(expectedCode);
    }

    it("should add at the end", () => {
      doTest(
        "function identifier();\nfunction identifier() {}\n",
        { returnType: "number" },
        "function identifier();\nfunction identifier(): number;\nfunction identifier() {}\n",
      );
    });
  });

  describe(nameof<FunctionDeclaration>("remove"), () => {
    function doTest(text: string, index: number, expectedText: string) {
      const { sourceFile } = getInfoFromText(text);
      sourceFile.getFunctions()[index].remove();
      expect(sourceFile.getFullText()).to.equal(expectedText);
    }

    function doOverloadTest(text: string, index: number, overloadIndex: number, expectedText: string) {
      const { sourceFile } = getInfoFromText(text);
      sourceFile.getFunctions()[index].getOverloads()[overloadIndex].remove();
      expect(sourceFile.getFullText()).to.equal(expectedText);
    }

    it("should remove the function declaration", () => {
      doTest("function I() {}\n\nfunction J() {}\n\nfunction K() {}", 1, "function I() {}\n\nfunction K() {}");
    });

    it("should remove only the specified function in ambient contexts (not all overloads)", () => {
      doTest("declare function test(): void;\ndeclare function test(): void;", 1, "declare function test(): void;\n");
    });

    it("should remove the function declaration and its overloads", () => {
      doTest("function I() {}\n\nfunction J(): void;\nfunction J() {}\n\nfunction K() {}", 1, "function I() {}\n\nfunction K() {}");
    });

    it("should remove the function declaration overload when the first", () => {
      doOverloadTest(
        "function I() {}\n\nfunction J(): void;\nfunction J() {}\n\nfunction K() {}",
        1,
        0,
        "function I() {}\n\nfunction J() {}\n\nfunction K() {}",
      );
    });

    it("should remove the function declaration overload when the middle", () => {
      doOverloadTest(
        "function I() {}\n\nfunction J(first): void;\nfunction J(second): void;\nfunction J(third): void;\nfunction J() {}\n\nfunction K() {}",
        1,
        1,
        "function I() {}\n\nfunction J(first): void;\nfunction J(third): void;\nfunction J() {}\n\nfunction K() {}",
      );
    });
    it("should remove the function declaration overload when last", () => {
      doOverloadTest(
        "function I() {}\n\nfunction J(first): void;\nfunction J(second): void;\nfunction J() {}\n\nfunction K() {}",
        1,
        1,
        "function I() {}\n\nfunction J(first): void;\nfunction J() {}\n\nfunction K() {}",
      );
    });
  });

  describe(nameof<FunctionDeclaration>("set"), () => {
    function doTest(startingCode: string, structure: OptionalKind<FunctionDeclarationSpecificStructure>, expectedCode: string) {
      const { sourceFile } = getInfoFromText(startingCode);
      sourceFile.getFunctions()[0].set(structure);
      expect(sourceFile.getText()).to.equal(expectedCode);
    }

    it("should not modify anything if the structure doesn't change anything", () => {
      const code = `funciton identifier(): string;\nfunction identifier() {\n}`;
      doTest(code, {}, code);
    });

    it("should replace overloads when changed", () => {
      const structure: OptionalKindAndTrivia<MakeRequired<FunctionDeclarationSpecificStructure>> = {
        overloads: [{ returnType: "string" }],
      };
      doTest(
        "function identifier(p): number;\nfunction identifier() {\n}",
        structure,
        "function identifier(): string;\nfunction identifier() {\n}",
      );
    });

    it("should remove overloads when specifying an empty array", () => {
      const structure: OptionalKindAndTrivia<MakeRequired<FunctionDeclarationSpecificStructure>> = {
        overloads: [],
      };
      doTest(
        "function identifier(p): number;\nfunction identifier() {\n}",
        structure,
        "function identifier() {\n}",
      );
    });
  });

  describe(nameof<FunctionDeclaration>("getStructure"), () => {
    function doTest(text: string, expectedStructures: OptionalProperties<OptionalTrivia<MakeRequired<FunctionDeclarationStructure>>, "overloads">[]) {
      const { sourceFile } = getInfoFromText<any>(text);
      const functionDecs = sourceFile.getFunctions();
      expect(functionDecs.length).to.equal(expectedStructures.length);

      for (let i = 0; i < expectedStructures.length; i++) {
        const expectedStructure = expectedStructures[i];
        const functionDec = functionDecs[i];
        const structure = functionDec.getStructure() as FunctionDeclarationStructure;
        expect(structure).to.deep.equal(fillStructures.functionDeclaration(expectedStructure));
      }
    }

    it("should get the structure for an empty function", () => {
      doTest("declare function test() {}", [{
        kind: StructureKind.Function,
        statements: [],
        docs: [],
        hasDeclareKeyword: true,
        isAsync: false,
        isDefaultExport: false,
        isExported: false,
        isGenerator: false,
        name: "test",
        overloads: [],
        parameters: [],
        returnType: undefined,
        typeParameters: [],
      }]);
    });

    it("should return two function structures for the same name when there is no implementation", () => {
      const structure: OptionalProperties<OptionalTrivia<MakeRequired<FunctionDeclarationStructure>>, "overloads"> = {
        kind: StructureKind.Function,
        statements: undefined,
        docs: [],
        hasDeclareKeyword: true,
        isAsync: false,
        isDefaultExport: false,
        isExported: false,
        isGenerator: false,
        name: "test",
        parameters: [],
        returnType: "void",
        typeParameters: [],
      };

      const structures = [structure, structure];
      doTest("declare function test(): void;\ndeclare function test(): void;", structures);
    });

    it("should get the structure for a function that has everything", () => {
      const code = `
/** docs2 */
export default async function *test<U>(p): number;
/** docs */
export default async function *test<T>(param): string {
    return '';
}
`;
      const overloadStructure: OptionalTrivia<MakeRequired<FunctionDeclarationOverloadStructure>> = {
        kind: StructureKind.FunctionOverload,
        docs: [{ description: "docs2" }],
        hasDeclareKeyword: false,
        isAsync: true,
        isDefaultExport: true,
        isExported: true,
        isGenerator: true,
        parameters: [{ name: "p" }],
        returnType: "number",
        typeParameters: [{ name: "U", variance: TypeParameterVariance.None }],
      };

      doTest(code, [{
        kind: StructureKind.Function,
        statements: ["return '';"],
        docs: [{ description: "docs" }],
        hasDeclareKeyword: false,
        isAsync: true,
        isDefaultExport: true,
        isExported: true,
        isGenerator: true,
        name: "test",
        overloads: [overloadStructure],
        parameters: [{ name: "param" }],
        returnType: "string",
        typeParameters: [{ name: "T", variance: TypeParameterVariance.None }],
      }]);
    });
  });
});
